Subtracting complex numbers in Python